package com.sky.service.impl;

import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.google.common.collect.Maps;
import com.sky.constant.MessageConstant;
import com.sky.constant.StatusConstant;
import com.sky.dto.DishDTO;
import com.sky.dto.DishPageQueryDTO;
import com.sky.entity.Dish;
import com.sky.entity.DishFlavor;
import com.sky.entity.Setmeal;
import com.sky.exception.BaseException;
import com.sky.exception.DeletionNotAllowedException;
import com.sky.exception.DishNotFoundException;
import com.sky.mapper.DishFlavorMapper;
import com.sky.mapper.DishMapper;
import com.sky.mapper.SetmealDishMapper;
import com.sky.mapper.SetmealMapper;
import com.sky.result.PageResult;
import com.sky.service.DishService;
import com.sky.vo.DishVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import java.util.stream.Collectors;


/**
 * ClassName: DishServiceImpl
 * Package: com.sky.service.impl
 * Description:
 *
 * @Auther gongkaiming
 * @Create 2024/7/28 13:37
 * @Version 1.0
 */
@Service
@Slf4j
public class DishServiceImpl implements DishService {

    @Autowired
    private DishMapper dishMapper;

    @Autowired
    private DishFlavorMapper dishFlavorMapper;

    @Autowired
    private SetmealDishMapper setmealDishMapper;

    @Autowired
    private SetmealMapper setmealMapper;

    @Transactional
    public void saveWithFlavor(DishDTO dishDTO) {

        Dish dish = new Dish();

        BeanUtils.copyProperties(dishDTO, dish);

        // 向菜品表插入数据
        dishMapper.insert(dish);
        // 获取insert语句生成得主键值
        Long dishId = dish.getId();

        List<DishFlavor> flavors = dishDTO.getFlavors();
        if (flavors != null && flavors.size() > 0) {
            flavors.forEach(dishFlavor -> {
                dishFlavor.setDishId(dishId);
            });

            // 向口味表插入n个数据
            dishFlavorMapper.insertBatch(flavors);
        }
    }

    /**
     * 菜品分页查询
     *
     * @param dishPageQueryDTO
     * @return
     */
    public PageResult pageQuery(DishPageQueryDTO dishPageQueryDTO) {
        PageHelper.startPage(dishPageQueryDTO.getPage(), dishPageQueryDTO.getPageSize());
        Page<DishVO> page = dishMapper.pageQuery(dishPageQueryDTO);

        return new PageResult(page.getTotal(), page.getResult());
    }

    /**
     * 批量删除
     *
     * @param ids
     */
    @Transactional
    public void deleteBatch(List<Long> ids) {

        // 删除当前菜品----是否在起售中？？
        for (Long id : ids) {
            Dish dish = dishMapper.getById(id);
            if (dish.getStatus() == StatusConstant.ENABLE) {
                // 处于起售中，不能删除
                throw new DeletionNotAllowedException(MessageConstant.DISH_ON_SALE);
            }
        }
        // 是否关联套餐
        List<Long> setmealIds = setmealDishMapper.getSetMealIdsByDishIds(ids);
        if (setmealIds != null && setmealIds.size() > 0) {
            // 当前菜品被关联了套餐，不能删除
            throw new DeletionNotAllowedException(MessageConstant.DISH_BE_RELATED_BY_SETMEAL);
        }
        // 删除菜品
//        for (Long id : ids) {
//            dishMapper.deleteById(id);
//            // 删除菜品关联口味
//            dishFlavorMapper.deleteByDishId(id);
//        }
        // 删除菜品代码优化
        // 1.根据菜品id集合批量删除菜品
        dishMapper.deleteByIds(ids);
        // 2.根据菜品id集合批量删除关联的口味数据
        dishFlavorMapper.deleteByDishIds(ids);
    }

    /**
     * 根据菜品id查询菜品和对应的口味数据
     *
     * @param id
     * @return
     */
    public DishVO getByIdWithFlavor(Long id) {
        // 根据id查询菜品数据
        Dish dish = dishMapper.getById(id);
        // 如果菜品不存在抛出不存在异常
        if (Objects.isNull(dish)) {
            throw new DishNotFoundException(MessageConstant.DISH_NOT_FOUND);
        }
        // 根据菜品id查询口味数据
        List<DishFlavor> dishFlavors = dishFlavorMapper.getByDishId(id);
        // 将菜品数据和口味数据封装到DishVO
        DishVO dishVO = new DishVO();
        BeanUtils.copyProperties(dish, dishVO);
        // 判断菜品中是否存在口味数据
        if (Objects.nonNull(dishFlavors) && !dishFlavors.isEmpty()) {
            dishVO.setFlavors(dishFlavors);
        }

        return dishVO;
    }

    /**
     * 根据id修改菜品基本信息和对应的口味信息
     *
     * @param dishDTO
     */
    @Override
    public void updateWithFlavor(DishDTO dishDTO) {
        Dish dish = new Dish();
        BeanUtils.copyProperties(dishDTO, dish);
        // 修改菜品表的基本信息
        dishMapper.update(dish);
        // 删除原有的口味信息
        dishFlavorMapper.deleteByDishId(dishDTO.getId());
        // 插入口味信息
        List<DishFlavor> flavors = dishDTO.getFlavors();
        if (flavors != null && flavors.size() > 0) {
            flavors.forEach(dishFlavor -> {
                dishFlavor.setDishId(dishDTO.getId());
            });

            // 向口味表插入n个数据
            dishFlavorMapper.insertBatch(flavors);
        }
    }

    /**
     * 菜品起售与停售
     *
     * @param status
     * @param id
     */
    @Transactional
    public void startOrStop(Integer status, Long id) {
        Dish dish = Dish.builder()
                .id(id)
                .status(status)
                .build();
        dishMapper.update(dish);
        if (status == StatusConstant.DISABLE) {
            // 若是停售，则将包含菜品的套餐停售
            List<Long> dishId = new ArrayList<>();
            dishId.add(id);
            List<Long> setmealIds = setmealDishMapper.getSetMealIdsByDishIds(dishId);
            if (setmealIds != null && setmealIds.size() > 0) {
                for (Long setmealId : setmealIds) {
                    Setmeal setmeal = Setmeal.builder()
                            .id(setmealId)
                            .status(StatusConstant.DISABLE)
                            .build();
                    setmealMapper.update(setmeal);
                }
            }

        }
    }

    /**
     * 根据分类id查询菜品
     *
     * @param categoryId
     * @return
     */
    public List<Dish> list(Long categoryId) {
        Dish dish = Dish.builder()
                .categoryId(categoryId)
                .status(StatusConstant.ENABLE)
                .build();
        return dishMapper.list(dish);
    }


    /**
     * 条件查询菜品和口味
     *
     * @param dish
     * @return
     */
    public List<DishVO> listWithFlavor(Dish dish) {
        List<Dish> dishList = dishMapper.list(dish);

        List<DishVO> dishVOList = new ArrayList<>();
        List<Long> dishIds = new ArrayList<>();
        Map<Long, Dish> dishMap;
        if (!dishList.isEmpty()) {
            dishIds = dishList.stream().map(Dish::getId).toList();
            dishMap = dishList.stream().collect(Collectors.toMap(Dish::getId, dto -> dto));
        } else {
            dishMap = Maps.newHashMap();
        }

        List<DishFlavor> dishFlavors = dishFlavorMapper.getByDishIds(dishIds);

        Map<Long, DishFlavor> dishFlavorHashMap = dishFlavors.stream()
                .collect(Collectors.toMap(DishFlavor::getDishId, dto -> dto));

        dishIds.forEach(dishId -> {
            DishVO dishVO = new DishVO();
            DishFlavor dishFlavor = dishFlavorHashMap.get(dishId);
            if (Objects.nonNull(dishFlavor)) {
                dishVO.setFlavors(dishFlavors);
            }
            Dish dish1 = dishMap.get(dishId);
            if (Objects.nonNull(dish1)) {
                BeanUtils.copyProperties(dish1, dishVO);
            }
            dishVOList.add(dishVO);
        });

//        for (Dish d : dishList) {
//            DishVO dishVO = new DishVO();
//
//
//            //根据菜品id查询对应的口味
//            List<DishFlavor> flavors = dishFlavorMapper.getByDishId(d.getId());
//
//            dishVO.setFlavors(flavors);
//            dishVOList.add(dishVO);
//        }

        return dishVOList;
    }


}
